import json
from country_codes import get_country_code
import pygal_maps_world.maps
from pygal.style import RotateStyle
from pygal.style import LightColorizedStyle

filename = 'Chapter16-Downloading-Data\population_json.json'

with open(filename) as f:
    pop_data = json.load(f)

cc_population = {}
for pop_dict in pop_data:
    if pop_dict['Year'] == 2010:
        country_name = pop_dict['Country Name']
        population = int(float(pop_dict['Value']))
        code = get_country_code(country_name)
        if code:
          cc_population[code] = population

cc_pops_1, cc_pops_2, cc_pops_3 = {}, {}, {}
for cc, pop in cc_population.items():
    if pop < 10000000:  # 少于 1000 万
        cc_pops_1[cc] = pop
    elif pop < 1000000000:  # 1000 万到 10 亿
        cc_pops_2[cc] = pop
    else:  # 超过 10 亿
        cc_pops_3[cc] = pop
print(len(cc_pops_1), len(cc_pops_2), len(cc_pops_3))
wm_stytle = RotateStyle("#994E33", base_style=LightColorizedStyle)

wm = pygal_maps_world.maps.World(style=wm_stytle)
wm._title = 'World Population in 2010, by Country'
wm.add('1-10m', cc_pops_1)
wm.add('10m-1bn', cc_pops_2)
wm.add('>1bn', cc_pops_3) 
wm.add('2010', cc_population)

wm.render_to_file('world_population.svg')
